cmake_minimum_required(VERSION 3.12)
project(GeluOperator)

set(CMAKE_CXX_STANDARD 17)

# Ascend环境配置
set(ASCEND_PATH /usr/local/Ascend/ascend-toolkit/latest)
include_directories(${ASCEND_PATH}/runtime/include)
link_directories(${ASCEND_PATH}/runtime/lib64/stub)

add_executable(gelu_op
    src/gelu_operator.cpp
    src/main.cpp
)

target_link_libraries(gelu_op
    ascendcl
    acl_op_compiler
    runtime
    hccl
)